package com.skiving.cloud.admin.controller;

import cn.dev33.satoken.annotation.SaCheckPermission;
import com.mybatisflex.core.paginate.Page;
import com.mybatisflex.core.query.QueryWrapper;
import com.skiving.cloud.admin.entity.*;
import com.skiving.cloud.admin.service.SysMenuService;
import com.skiving.cloud.admin.service.SysRoleMenuService;
import com.skiving.cloud.admin.service.SysUserRoleService;
import com.skiving.cloud.auth.util.UserUtil;
import com.skiving.cloud.common.exception.BusinessException;
import com.skiving.cloud.common.utils.PageParameter;
import com.skiving.cloud.common.utils.ReturnT;
import com.skiving.cloud.common.utils.UuidUtil;
import com.skiving.cloud.constant.AdminConstant;
import com.skiving.cloud.logutils.OperateLogLocal;
import lombok.extern.slf4j.Slf4j;
import org.noear.solon.Utils;
import org.noear.solon.annotation.*;
import org.noear.solon.core.util.MimeType;
import com.skiving.cloud.admin.service.SysRoleService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;

import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 角色 控制层。
 *
 * @author skivingcloud
 * @since 2025-03-08
 */
@Slf4j
@Controller
@Api("角色接口")
@Mapping("/sysRole")
@Produces(MimeType.APPLICATION_JSON_UTF8_VALUE)
public class SysRoleController {

    @Inject
    private SysRoleService sysRoleService;
    @Inject
    private SysMenuService sysMenuService;
    @Inject
    private SysRoleMenuService sysRoleMenuService;
    @Inject
    private SysUserRoleService sysUserRoleService;
    /**
     * 添加角色。
     *
     * @param sysRole 角色
     * @return {@code true} 添加成功，{@code false} 添加失败
     */
    @Post
    @Mapping("/save")
    @ApiOperation("保存角色")
    @OperateLogLocal(module="系统管理-角色管理", description = "新增角色")
    @SaCheckPermission("sys:role:add")
    public ReturnT<Boolean> save(@Body @ApiParam("角色") SysRole sysRole) {
        sysRole.setId(UuidUtil.get32UUID());
        sysRole.setCreateUserId(UserUtil.getCurrentUser().getId());
        sysRole.setCreateTime(LocalDateTime.now());
        return ReturnT.ok(sysRoleService.save(sysRole));
    }

    /**
     * 根据主键删除角色。
     *
     * @param id 主键
     * @return {@code true} 删除成功，{@code false} 删除失败
     */
    @Get
    @Mapping("/remove/{id}")
    @ApiOperation("根据主键角色")
    @OperateLogLocal(module="系统管理-角色管理", description = "删除角色")
    @SaCheckPermission("sys:role:delete")
    public ReturnT<Boolean> remove(@Path @ApiParam("角色主键") String id) {
        return ReturnT.ok(sysRoleService.removeById(id));
    }

    /**
     * 根据主键更新角色。
     *
     * @param sysRole 角色
     * @return {@code true} 更新成功，{@code false} 更新失败
     */
    @Post
    @Mapping("/update")
    @ApiOperation("根据主键更新角色")
    @OperateLogLocal(module="系统管理-角色管理", description = "修改角色")
    @SaCheckPermission("sys:role:update")
    public ReturnT<Boolean> update(@Body @ApiParam("角色主键") SysRole sysRole) {
        sysRole.setModUserId(UserUtil.getCurrentUser().getId());
        sysRole.setModTime(LocalDateTime.now());
        return ReturnT.ok(sysRoleService.updateById(sysRole));
    }

    /**
     * 根据角色主键获取详细信息。
     *
     * @param id 角色主键
     * @return 角色详情
     */
    @Get
    @Mapping("/getInfo/{id}")
    @ApiOperation("根据主键获取角色")
    public ReturnT<SysRole> getInfo(@Path @ApiParam("角色主键") String id) {
        return ReturnT.ok(sysRoleService.getById(id));
    }

    /**
     * 分页查询角色。
     *
     * @param parameter 分页对象
     * @return 分页对象
     */
    @Post
    @Mapping("/page")
    @ApiOperation("分页查询角色")
    public ReturnT<Page<SysRole>> page(@Body @ApiParam("分页信息") PageParameter<SysRole> parameter) {
        return ReturnT.ok(sysRoleService.queryByPage(parameter));
    }

    /**
     * 分页查询角色
     * @param parameter 查询参数
     * @return 分页查询结果
     */
    @ApiOperation("角色列表")
    @Post
    @Mapping("/queryRoleListPage")
    public ReturnT<Page<SysRole>> queryRoleListPage(@Body PageParameter<SysRole> parameter){
        Page<SysRole> page = sysRoleService.queryByPage(parameter);
        return ReturnT.ok(page);
    }

    /**
     * 根据租户和角色id加载资源树权限
     *
     * @param tenantId 租户ID
     * @param id 角色ID
     * @return 资源权限
     */
    @ApiOperation("根据租户和角色id加载资源树权限")
    @Get
    @Mapping("/resetTenantRoleResTree/{tenantId}/{id}")
    public ReturnT<Map<String, Object>> resetTenantRoleResTree(@Path("tenantId") String tenantId, @Path("id") String id){
        if (!UserUtil.getCurrentUser().getTenantId().equals(AdminConstant.SUPER_ADMIN_TENANT_ID)) {
            tenantId = UserUtil.getCurrentUser().getTenantId();
        }
        if (tenantId == null) {
            return ReturnT.error("租户不明确没数据");
        }
        List<SysMenu> list = sysMenuService.getTenantMenuList(tenantId);
        List<SysRoleMenu> asingRes = new ArrayList<>();
        if(Utils.isNotBlank(id)){
            QueryWrapper queryWrapper = QueryWrapper.create();
            queryWrapper.where(SysRoleMenu::getRoleId).eq(id);
            asingRes = sysRoleMenuService.list(queryWrapper);
        }
        Map<String, Object> res = new HashMap<>();
        res.put("tree", list);
        res.put("asingRes", asingRes);
        return ReturnT.ok(res);
    }

    /**
     * 设置角色资源权限
     * @param sysRole 角色资源信息
     * @return 结果
     */
    @ApiOperation("设置角色资源权限")
    @Post
    @Mapping("/setRoleMenu")
    @OperateLogLocal(module="系统管理-角色管理", description = "设置角色资源权限")
    @SaCheckPermission("sys:role:setmenu")
    public ReturnT<String> setRoleMenu(@Body SysRole sysRole) {
        try {
            sysRoleService.saveRoleRes(sysRole.getId(), sysRole.getResId());
            return ReturnT.ok();
        } catch (Exception e) {
            log.error("设置资源权限失败", e);
            return ReturnT.error("设置资源权限失败");
        }
    }

    /**
     * 查询用户角色关系
     * @param parameter 参数
     * @return 结果
     */
    @ApiOperation("分页查询角色下的用户")
    @Post
    @Mapping("/getRoleUserDataListByRoleId")
    public ReturnT<Page<SysUserRole>> getRoleUserDataListByRoleId(@Body PageParameter<SysUserRole> parameter){
        if(parameter.getParam().getRoleId().isEmpty()){
            return ReturnT.error("必须选择一条角色记录");
        }
        Page<SysUserRole> page = sysUserRoleService.queryByPage(parameter);
        return ReturnT.ok(page);
    }

    /**
     * 查询没设置该角色的用户
     * @param parameter 参数
     * @return 参数
     */
    @ApiOperation("查询没设置该角色的用户")
    @Post
    @Mapping("/getRoleUserSelectListByRoleId")
    public ReturnT<Page<SysUser>> getRoleUserSelectListByRoleId(@Body PageParameter<SysUserRole> parameter){
        if(parameter.getParam().getRoleId().isEmpty()){
            return ReturnT.error("必须选择一条角色记录");
        }
        Page<SysUser> page = sysUserRoleService.queryNotSetByPage(parameter);
        return ReturnT.ok(page);
    }

    /**
     * 删除角色用户关系
     * @param id 关系id
     * @return 删除结果
     */
    @ApiOperation("删除角色用户关系")
    @Get
    @Mapping("/deleteRoleUserById/{id}")
    @OperateLogLocal(module="系统管理-角色管理", description = "删除角色用户关系")
    @SaCheckPermission("sys:roleUser:delete")
    public ReturnT<String> deleteRoleUserById(@Path("id") String id) {
        try{
            SysUserRole sysUserRole = sysUserRoleService.getById(id);
            if(sysUserRole != null && AdminConstant.SUPER_ADMIN_ROLE_ID.equals(sysUserRole.getRoleId())
                    && AdminConstant.SUPER_ADMIN.equals(sysUserRole.getUserId())){
                return ReturnT.error("内置用户角色不允许删除");
            }
            sysUserRoleService.removeById(id);
            return ReturnT.ok("删除删除角色用户关系成功");
        }catch(IllegalArgumentException | BusinessException e){
            log.error(e.getMessage(), e);
            return ReturnT.error(e.getMessage());
        }catch(Exception e){
            log.error("删除删除角色用户关系失败", e);
            return ReturnT.error("删除删除角色用户关系失败");
        }
    }

    /**
     * 新增用户角色关系
     * @param sysUserRole 用户角色关系
     * @return 结果
     */
    @ApiOperation("新增用户角色关系")
    @Post
    @Mapping("/setRoleUserSelectHandle")
    @OperateLogLocal(module="系统管理-角色管理", description = "新增角色用户关系")
    @SaCheckPermission("sys:roleUser:add")
    public ReturnT<Boolean> setRoleUserSelectHandle(@Body SysUserRole sysUserRole) {
        try{
            sysUserRole.setId(UuidUtil.get32UUID());
            return ReturnT.ok(sysUserRoleService.save(sysUserRole));
        }catch(Exception e){
            log.error("保存新增用户角色关系失败", e);
            return ReturnT.error("保存新增用户角色关系失败！");
        }
    }

    /**
     * 获取当前用户租户下的角色
     * @return 角色信息
     */
    @ApiOperation("获取当前用户租户下的角色")
    @Get
    @Mapping("/getTenantRoleList")
    public ReturnT<List<SysRole>> getTenantRoleList(){
        return ReturnT.ok(sysRoleService.listByTenantId(UserUtil.getCurrentUser().getTenantId()));
    }
}
